import pandas as pd

# 文件路径
input_file = r'D:\xlzx\zhs\油统计.xlsx'
output_file = r'D:\xlzx\zhs\油统计结果.xlsx'

# 读取指定的两个Sheet
sheet_names = ['7.1-7.31', '8.1-8.31']

# 分别读取两个Sheet的数据
df_july = pd.read_excel(input_file, sheet_name=sheet_names[0])
df_august = pd.read_excel(input_file, sheet_name=sheet_names[1])

# 打印列名以确认
print("7.1-7.31 Sheet列名:", df_july.columns)
print("8.1-8.31 Sheet列名:", df_august.columns)

# 检查列名是否一致
assert '车号' in df_july.columns, "df_july缺少'车号'列"
assert '加油(升)' in df_july.columns, "df_july缺少'加油(升)'列"
assert '车号' in df_august.columns, "df_august缺少'车号'列"
assert '加油(升)' in df_august.columns, "df_august缺少'加油(升)'列"

# 合并两个DataFrame，按车号进行左连接
merged_df = pd.merge(df_july, df_august, on='车号', suffixes=('_july', '_august'))

# 计算总加油量
merged_df['总加油量'] = merged_df['加油(升)_july'].fillna(0) + merged_df['加油(升)_august'].fillna(0)

# 选择需要的列
result_df = merged_df[['车号', '总加油量']]

# 输出结果
print(result_df)

# 保存结果到新的Excel文件
result_df.to_excel(output_file, index=False)

print(f"结果已保存到 {output_file}")